/*
**    Copyright (C) 2003-2010 Institute for Systems Biology
**                            Seattle, Washington, USA.
**
**    This library is free software; you can redistribute it and/or
**    modify it under the terms of the GNU Lesser General Public
**    License as published by the Free Software Foundation; either
**    version 2.1 of the License, or (at your option) any later version.
**
**    This library is distributed in the hope that it will be useful,
**    but WITHOUT ANY WARRANTY; without even the implied warranty of
**    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
**    Lesser General Public License for more details.
**
**    You should have received a copy of the GNU Lesser General Public
**    License along with this library; if not, write to the Free Software
**    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307  USA
*/
package org.systemsbiology.google.visualization.datasource.jdbc;

import org.json.JSONException;
import org.json.JSONObject;
import org.springframework.dao.DataAccessException;
import org.springframework.dao.DataRetrievalFailureException;
import org.springframework.jdbc.core.ConnectionCallback;

import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;

/**
 * @author hrovira
 */
public class DatabaseTableColumnConnectionCallback implements ConnectionCallback<JSONObject> {
    private final String tableName;
    private String appendKey = "items";

    public DatabaseTableColumnConnectionCallback(String tableName) {
        this.tableName = tableName;
    }

    public DatabaseTableColumnConnectionCallback(String tableName, String appendKey) {
        this.tableName = tableName;
        this.appendKey = appendKey;
    }

    public JSONObject doInConnection(Connection connection) throws SQLException, DataAccessException {
        try {
            JSONObject json = new JSONObject();

            ResultSet rs = connection.getMetaData().getColumns(null, null, tableName, null);
            while (rs.next()) {
                JSONObject column = new JSONObject();
                column.put("name", rs.getString(4));
                column.put("datatype", rs.getString(6));
                json.append(appendKey, column);
            }

            return json;
        } catch (JSONException e) {
            throw new DataRetrievalFailureException(e.getMessage(), e);
        }
    }
}